﻿using System;
using System.IO.Pipes;
using System.Threading.Tasks;
using Harvester.Core.Messaging.Sources;
using Harvester.Core.Messaging.Sources.NamedPipe;
using Xunit;
using Xunit.Extensions;

/* Copyright (c) 2012-2013 CBaxter
 * 
 * Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), 
 * to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, 
 * and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
 * 
 * The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
 * 
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS 
 * IN THE SOFTWARE. 
 */

namespace Harvester.Core.Tests.Messaging.Sources.NamedPipe.UsingNamedPipeClientBuffer
{
    public class WhenWritingToOpenBuffer : IDisposable
    {
        private readonly NamedPipeServerStream serverPipeStream;
        private readonly MessageBuffer buffer;
        private readonly String pipeName;

        public WhenWritingToOpenBuffer()
        {
            pipeName = @"\\.\pipe\" + Guid.NewGuid();
            buffer = new NamedPipeClientBuffer(".", pipeName) { Timeout = TimeSpan.FromMilliseconds(250) };
            serverPipeStream = new NamedPipeServerStream(pipeName, PipeDirection.InOut);
        }

        public void Dispose()
        {
            buffer.Dispose();
        }

        [Fact]
        public void CanWriteNullMessage()
        {
            Assert.DoesNotThrow(() => buffer.Write(null));
        }

        [Fact]
        public void CanWriteEmptyMessage()
        {
            Assert.DoesNotThrow(() => buffer.Write(new Byte[0]));
        }
        
        [Fact]
        public void CanHandleLargeWritesVolume()
        {
            const Int32 iterations = 10000;
            const Int32 messageSize = 4096;
            var clients = new[]
                              {
                                  Task.Factory.StartNew(() => RunClient(iterations, messageSize), TaskCreationOptions.PreferFairness),
                                  Task.Factory.StartNew(() => RunClient(iterations, messageSize), TaskCreationOptions.PreferFairness),
                                  Task.Factory.StartNew(() => RunClient(iterations, messageSize), TaskCreationOptions.PreferFairness),
                                  Task.Factory.StartNew(() => RunClient(iterations, messageSize), TaskCreationOptions.PreferFairness),
                                  Task.Factory.StartNew(() => RunClient(iterations, messageSize), TaskCreationOptions.PreferFairness)
                              };

            var bytes = new Byte[messageSize];
            for (var i = 0; i < iterations * clients.Length; i++)
            {
                serverPipeStream.WaitForConnection();

                Assert.Equal(messageSize, serverPipeStream.Read(bytes, 0, bytes.Length));

                serverPipeStream.Disconnect();
            }

            Task.WaitAll(clients);
        }

        private void RunClient(Int32 iterations, Int32 messageSize)
        {
            for (var i = 0; i < iterations; i++)
                Assert.DoesNotThrow(() => buffer.Write(new Byte[messageSize]));
        }
    }
}
